# -*- coding: utf-8 -*-
# @Author: Tim Liu
# @Date: 2024-06-04
# @Last Modified by: Tim Liu
# @Last Modified time: 2024-06-04

# @Description: search request schema

from pydantic import BaseModel

class SearchRequest(BaseModel):
    """
        question: user question
        kbase_id: knowledge base id
        kbase_name: knowledge base name
        
        web_search : true(1) use web search to get more context, false(0) do not use web search
    """
    question: str = None
        
    kbase_id: int = 0
    
    kbase_name: str = None
    
    # search web for context
    web_search: int = 0